import os
import sys
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))

from mask.patch_image import patchimg
import cv2
import numpy as np
import matplotlib.pyplot as plt

if __name__ == '__main__':
    init_pose_path='plot_bigger/init/0288.jpg'
    gt_pose_path='plot_bigger/gt_0288.jpg'
    refine_pose_path='plot_bigger/result_0288.jpg'


    init_pose = cv2.imread(init_pose_path)
    gt_pose = cv2.imread(gt_pose_path)
    refine_pose = cv2.imread(refine_pose_path)

    init_pose = patchimg(init_pose, cx=141, cy=141, r=100)
    gt_pose = patchimg(gt_pose, cx=141, cy=141, r=100) 
    refine_pose = patchimg(refine_pose, cx=141, cy=141, r=100)
    cv2.imwrite('plot_bigger/0288_init_pose.jpg',init_pose)
    cv2.imwrite('plot_bigger/0288_gt_pose.jpg',gt_pose)
    cv2.imwrite('plot_bigger/0288_refine_pose.jpg',refine_pose)
